from pydantic import BaseModel
from typing import Any, Dict, List

class UserRequest(BaseModel):
    prompt: str
    
class DataResponse(BaseModel):
    status: str
    message: str
    data: List[Dict[str, Any]]
    prompt: str
    inferred_type: str
    count: int
